vim.keymap.set('n', ',pa', function()
    vim.cmd('split')
    vim.cmd('horizontal resize 7')
    vim.cmd.terminal('python -m pytest -s')
end, { buffer = 0, desc = 'Run pytest for all tests' })

vim.keymap.set('n', ',pf', function()
    vim.cmd('split')
    vim.cmd('horizontal resize 7')
    vim.cmd.terminal('python -m pytest % -s')
end, { buffer = 0, desc = 'Run pytest for current file' })
